package com.huishu.file.utils;

import org.apache.commons.codec.digest.DigestUtils;
import org.apache.commons.lang3.StringUtils;

import java.nio.charset.StandardCharsets;
import java.util.concurrent.ThreadLocalRandom;

/**
 * 高性能的创建UUID的工具类，https://github.com/lets-mica/mica
 *
 * @author yadong.zhang (yadong.zhang0415(a)gmail.com)
 * @since 1.9.3
 */
public class UuidUtils {

    /**
     * All possible chars for representing a number as a String
     * copy from mica：https://github.com/lets-mica/mica/blob/master/mica-core/src/main/java/net/dreamlu/mica/core/utils/NumberUtil.java#L113
     */
    private final static byte[] DIGITS = {
            '0', '1', '2', '3', '4', '5',
            '6', '7', '8', '9', 'a', 'b',
            'c', 'd', 'e', 'f', 'g', 'h',
            'i', 'j', 'k', 'l', 'm', 'n',
            'o', 'p', 'q', 'r', 's', 't',
            'u', 'v', 'w', 'x', 'y', 'z',
            'A', 'B', 'C', 'D', 'E', 'F',
            'G', 'H', 'I', 'J', 'K', 'L',
            'M', 'N', 'O', 'P', 'Q', 'R',
            'S', 'T', 'U', 'V', 'W', 'X',
            'Y', 'Z'
    };

    /**
     * 生成uuid，采用 jdk 9 的形式，优化性能
     * copy from mica：https://github.com/lets-mica/mica/blob/master/mica-core/src/main/java/net/dreamlu/mica/core/utils/StringUtil.java#L335
     * <p>
     * 关于mica uuid生成方式的压测结果，可以参考：https://github.com/lets-mica/mica-jmh/wiki/uuid
     *
     * @return UUID
     */
    public static String getUUID() {
        ThreadLocalRandom random = ThreadLocalRandom.current();
        long lsb = random.nextLong();
        long msb = random.nextLong();
        byte[] buf = new byte[32];
        formatUnsignedLong(lsb, buf, 20, 12);
        formatUnsignedLong(lsb >>> 48, buf, 16, 4);
        formatUnsignedLong(msb, buf, 12, 4);
        formatUnsignedLong(msb >>> 16, buf, 8, 4);
        formatUnsignedLong(msb >>> 32, buf, 0, 8);
        return new String(buf, StandardCharsets.UTF_8);
    }

    /**
     * copy from mica：https://github.com/lets-mica/mica/blob/master/mica-core/src/main/java/net/dreamlu/mica/core/utils/StringUtil.java#L348
     */
    private static void formatUnsignedLong(long val, byte[] buf, int offset, int len) {
        int charPos = offset + len;
        int radix = 1 << 4;
        int mask = radix - 1;
        do {
            buf[--charPos] = DIGITS[((int) val) & mask];
            val >>>= 4;
        } while (charPos > offset);
    }

    final static String[] chars = new String[]{"0", "1", "2", "3", "4", "5",

            "6", "7", "8", "9", "A", "C", "D", "E", "F", "G", "H",

            "J", "K", "M", "N", "P", "R", "S", "T",

            "U", "V", "W", "X", "Y", "Z"};

    public static String codeGenUtil(String id, Integer len) {
        if (null == id) {
            return null;
        }
        int index = 0;
        String code = null;

        do {
            code = genInviteCode(id, len);

            if (!StringUtils.isNumeric(code)) {
                return code;
            }
            id = code + id + System.nanoTime();
            index++;
        } while (index < 20);
        if (StringUtils.isNumeric(code)) {
            return codeGenUtil(UuidUtils.getUUID(), 6);
        }
        return code;
    }

    public static String genInviteCode(Integer len) {
        return codeGenUtil(getUUID(), len);
    }

    //传入的加密密钥id必须唯一的，比如用户id、用户ip，我这是优惠券id
    //参数len 是你要生成该随机码的长度，要生成5位的就传5
    private static String genInviteCode(String id, Integer len) {
        // 对传入id进行 MD5 加密，生成32位字符串，是唯一的
        String sMD5EncryptResult = DigestUtils.md5Hex(id + System.currentTimeMillis());
        StringBuilder stringBuilder = new StringBuilder();
        Integer subLen = sMD5EncryptResult.length() / len;
        for (int i = 0; i < len; i++) {
            // 把加密字符按照 sublen 位一组 16 进制与 0x3FFFFFFF（30位1） 进行位与运算（作用：将生成的long型数据控制在30位）
            String sTempSubString = sMD5EncryptResult.substring(i * subLen, (i + 1) * subLen);
            // 这里需要使用 long 型来转换，因为 Inteper .parseInt() 只能处理 31 位 , 首位为符号位 , 如果不用
            // long ，则会越界与 0x3FFFFFFF（30位1） 进行位与运算（作用：因为long是64位，将生成的long型数据控制在30位）
            long lHexLong = 0x3FFFFFFF & Long.parseLong(sTempSubString, 16);//

            // 把得到的值与 0x0000001f 进行位与运算（作用，0x0000001f是16进制，转化为10进制为31，二进制00001111，这个位运算作用就是为了将index控制在31以内）
            // 取得字符数组 chars 索引//
            Long index = 0x0000001f & lHexLong;
            if (index.compareTo(Long.valueOf(chars.length)) >= 0) {
                index = Long.valueOf(chars.length - 1);
            }
            // 把取得的字符相加
            String outChars = chars[index.intValue()];
            // 把字符串存入对应索引的输出数组
            stringBuilder.append(outChars);
        }
        return stringBuilder.toString();
    }
}
